import torch
import torchvision

vgg16 = torchvision.models.vgg16()

#保存网络模型的两种方式，一种为将模型本身保存，一种为将模型对应参数保存
#方式1
torch.save(vgg16,"vgg16_model1.pth")
#对应导入方法
model = torch.load("vgg16_model1.pth")

#方式2
torch.save(vgg16.state_dict(),"vgg16_model2.pth")

#对应导入方法
vgg16_1 = torchvision.models.vgg16()
vgg16_1.load_state_dict(torch.load("vgg16_model2.pth"))
